package cn.xinfei.xdecision.datasource.mysql;

import cn.hutool.core.date.DateUtil;
import cn.xinfei.xdecision.common.model.BaseDO;
import cn.xinfei.xdecision.common.model.login.SessionManager;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.util.Objects;

/**
 * 通用参数填充实现类
 * <p>
 * 如果没有显式的对通用参数进行赋值，这里会对通用参数进行填充、赋值
 *
 * @author hexiaowu
 */
@Component
@Slf4j
public class DefaultDBFieldHandler implements MetaObjectHandler {

    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("insertFill start.");
        if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseDO) {
            BaseDO baseDO = (BaseDO) metaObject.getOriginalObject();

            String current = DateUtil.now();
            if (StringUtils.isEmpty(baseDO.getCreatedTime())) {
                baseDO.setCreatedTime(current);
            }

            if (StringUtils.isEmpty(baseDO.getUpdatedTime())) {
                baseDO.setUpdatedTime(current);
            }

            Long userId = SessionManager.getLoginUserId();
            // 当前登录用户不为空，则当前登录用户为创建人
            if (Objects.nonNull(userId)) {
                if (Objects.isNull(baseDO.getCreateUserId())) {
                    baseDO.setCreateUserId(userId);
                }
                // 当前登录用户不为空，则当前登录用户为更新人
                if (Objects.isNull(baseDO.getUpdateUserId())) {
                    baseDO.setUpdateUserId(userId);
                }
            }
        }
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        // 更新时间为空，则以当前时间为更新时间
        Object modifyTime = getFieldValByName("updatedTime", metaObject);
        if (Objects.isNull(modifyTime)) {
            setFieldValByName("updatedTime", DateUtil.now(), metaObject);
        }

        // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
//        Object modifier = getFieldValByName("updateUserId", metaObject);
        Long userId = SessionManager.getLoginUserId();
        if (Objects.nonNull(userId)) {
            setFieldValByName("updateUserId", userId, metaObject);
        }
    }
}
